Papers with one-layer transformer
RetroMAE: Pre-Training Retrieval-oriented Language Models Via Masked Auto-Encoder (2022.emnlp-main)
Copied to clipboard
| Challenge: | Existing methods for dense retrieval are not effective, but there are still challenges. |
| Approach: | They propose a retrieval oriented pre-training paradigm based on Masked Auto-Encoder (MAE) where the sentence embedding is generated from the encoder’s masked input and the original sentence is recovered based upon the sentence embedded and decoded input via mangled language modeling. |
| Outcome: | The proposed model significantly improves the SOTA performance on a wide range of NLP benchmarks, like BEIR and MS MARCO. |